package com.yzy.fly.config;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.yzy.fly.util.DateUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;
import java.util.Date;

@Slf4j
@Component
public class MyMetaObjectHandler implements MetaObjectHandler {

    @Override
    public void insertFill(MetaObject metaObject) {
        log.info("start insert fill ....");
//        this.strictInsertFill(metaObject, "createTime", LocalDateTime.class, LocalDateTime.now()); // 起始版本 3.3.0(推荐使用)
        // 或者
        //先判断是否存在该字段
        boolean createTime = metaObject.hasSetter("createTime");
        if (createTime) {
            this.strictInsertFill(metaObject, "createTime", () -> DateUtil.getCurrentDate(), Date.class); // 起始版本 3.3.3(推荐)
        }

        boolean updateTime = metaObject.hasSetter("updateTime");
        if (updateTime) {
            this.strictInsertFill(metaObject, "updateTime", () -> DateUtil.getCurrentDate(), Date.class); // 起始版本 3.3.3(推荐)
        }
        // 或者
//        this.fillStrategy(metaObject, "createTime", LocalDateTime.now()); // 也可以使用(3.3.0 该方法有bug)
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        log.info("start update fill ....");
//        this.strictUpdateFill(metaObject, "updateTime", LocalDateTime.class, LocalDateTime.now()); // 起始版本 3.3.0(推荐)
        // 或者
        boolean updateTime = metaObject.hasSetter("updateTime");
        if (updateTime) {
//            this.strictUpdateFill(metaObject, "updateTime", () -> DateUtil.getCurrentDate(), Date.class); // 起始版本 3.3.3(推荐)
            /**
             * 更新时填充策略会判断属性已有值，不进行自动填充，因此update_time不会自动更新。
             * 而且官方目前没有直接提供强制更新的方法。
             * 填充策略,默认有值不覆盖,如果提供的值为null也不填充
             */
            this.setFieldValByName("updateTime", DateUtil.getCurrentDate(), metaObject);
        }
        // 或者
//        this.fillStrategy(metaObject, "updateTime", LocalDateTime.now()); // 也可以使用(3.3.0 该方法有bug)
    }
}